#pragma once
#include "YoloDet.h"

class DetectPlate
{
public:
    DetectPlate();
    ~DetectPlate();
    int init();
    int get_best_reslut(cv::Mat img, YoloDetection& best_reslut) ;
protected:
    YoloDet detecter;
    std::string modle_path = "chargeRobot/model/GreenPlateDet.onnx";
    std::string class_path = "chargeRobot/model/classes1.txt";

    float score_thre = 0.2f;			//类别阈值
    float nms_thre = 0.8f;              //NMS阈值，IOU 大于该值舍去，越大舍去的越少
    float conf_thre = 0.2f;             //置信度
};

